1607D - Blue-Red Permutation - CodeForces Solution


greedy math sortings *1300

Please click on ads to support us..

C++ Code:

#include<bits/stdc++.h>
#define int long long 
using namespace std;

int32_t main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);

    int t;
    cin>>t;
    while(t--)
    {
        int n;
        cin>>n;
        int a[n];
        for(int i=0;i<n;i++)
        {
            cin>>a[i];
        }
        string s;
        cin>>s;
        int red=0,blue=0;
        for(int i=0;i<n;i++)
        {
            if(s[i]=='R')
            {
                red++;
            }
            else
            {
                blue++;
            }
        }
        vector<pair<char,int>>vec(n);
        for(int i=0;i<n;i++)
        {
            vec[i]={s[i],a[i]};
        }
        bool b=0;
        for(int i=0;i<n;i++)
        {
            if(a[i]>n && s[i]=='R')
            {
                b=1;
                break;
            }
            else if(a[i]<=0 && s[i]=='B')
            {
                b=1;
                break;
            }
        }
        if(b==1)
        {
            cout<<"NO"<<'\n';
        }
        else
        {
            sort(vec.begin(),vec.end());
        
            for(int j=0;j<blue;j++)
            {
                if(vec[j].second<j+1)
                {
                    b=1;
                    break;
                }
            }
            for(int j=blue;j<n;j++)
            {
                if(vec[j].second>j+1)
                {
                    b=1;
                    break;
                }
            }
            if(b==0)
            {
                cout<<"YES"<<'\n';
            }
            else
            {
                cout<<"NO"<<'\n';
            }
        }
    }
}


Comments

Submit
0 Comments
More Questions

429. N-ary Tree Level Order Traversal
739. Daily Temperatures
647. Palindromic Substrings
583. Delete Operation for Two Strings
518. Coin Change 2
516. Longest Palindromic Subsequence
468. Validate IP Address
450. Delete Node in a BST
445. Add Two Numbers II
442. Find All Duplicates in an Array
437. Path Sum III
436. Find Right Interval
435. Non-overlapping Intervals
406. Queue Reconstruction by Height
380. Insert Delete GetRandom O(1)
332. Reconstruct Itinerary
368. Largest Divisible Subset
377. Combination Sum IV
322. Coin Change
307. Range Sum Query - Mutable
287. Find the Duplicate Number
279. Perfect Squares
275. H-Index II
274. H-Index
260. Single Number III
240. Search a 2D Matrix II
238. Product of Array Except Self
229. Majority Element II
222. Count Complete Tree Nodes
215. Kth Largest Element in an Array